#ifndef BATTERY_H
#define BATTERY_H

#include <string>
#include <iostream>
#include <fstream>

class Battery {
public:
    std::string model = "18650";
    std::string capacity = "48V 20Ah";
    std::string charge_time = "4h";

    // Set methods
    void setModel(const std::string& model);
    void setCapacity(const std::string& capacity);
    void setChargeTime(const std::string& charge_time);

    // Print and save methods
    void print() const;
    void save(std::ofstream& out) const;
};

#endif // BATTERY_H